package com.hccake.ballcat.codegen.util;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;

import javax.mail.*;
import javax.mail.internet.InternetAddress;
import javax.mail.internet.MimeMessage;
import javax.mail.internet.MimeMultipart;
import javax.mail.internet.MimeUtility;
import javax.mail.search.*;
import java.io.*;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.Properties;

@Slf4j
public class ReadTencentMail {

	public static void main(String[] args) throws Exception {
		readEmailPop();
	}

	public static void readEmailPop() throws Exception {
		Properties props = new Properties();
		props.put("mail.pop3.host", "pop.exmail.qq.com");
		props.put("mail.pop3.auth", "true");
		props.setProperty("mail.store.protocol", "pop3");
		props.put("mail.pop3.starttls.enable", "true");
		Session session = Session.getInstance(props);
		Store store = session.getStore("pop3");
		store.connect("pop.exmail.qq.com", "cheng_cd@murongtech.com", "evgiqDLMDHMX9Mhq");
		Folder folder = store.getFolder("Inbox");
		// 邮箱打开方式
		folder.open(Folder.READ_WRITE);
		// 昨天零点时间
		Calendar cal = Calendar.getInstance();
		cal.add(Calendar.DATE, -30);
		cal.set(Calendar.HOUR_OF_DAY, 0);
		cal.set(Calendar.MINUTE, 0);
		cal.set(Calendar.SECOND, 0);
		cal.set(Calendar.MILLISECOND, 0);
		Date mondayDate = cal.getTime();
		// 昨天零点到当前时间的邮件
		SearchTerm comparisonTermGe = new SentDateTerm(ComparisonTerm.GE, mondayDate);
		SearchTerm comparisonTermLe = new SentDateTerm(ComparisonTerm.LE, new Date());

		// SearchTerm titleCom = new SubjectTerm("机端接入待关注确认");
		SearchTerm comparisonAndTerm = new AndTerm(new SearchTerm[] { comparisonTermGe, comparisonTermLe });
		// 条件查询只能通过pop协议的方式才生效
		Message[] messages = folder.search(comparisonAndTerm);
		// Message[] messages = folder.getMessages();
		log.info("读取的邮件总数: " + messages.length);
		parseFileMessage(messages);
		// folder.setFlags(messages, new Flags(Flags.Flag.SEEN), true);
		log.info("邮件解析任务执行完毕");
	}

	public static void parseFileMessage(Message... messages) throws Exception {
		if (messages == null || messages.length < 1) {
			log.info("没有可读取邮件");
			return;
		}
		// 解析读取到的邮件 <23B689F502DF7F9F+2024112811211227865010@murongtech.com>
		for (Message message : messages) {
			MimeMessage msg = (MimeMessage) message;
			System.out.println("xxx" + msg.getMessageID());
			log.info("------------------解析第" + msg.getMessageNumber() + "封邮件-------------------- ");
			log.info("主题: " + MimeUtility.decodeText(msg.getSubject()));
			log.info("发件人: " + getFrom(msg));
			log.info("收件人：" + getReceiveAddress(msg, null));
			log.info("发送时间：" + getSentDate(msg, null));
			log.info("是否已读：" + isSeen(msg));
			log.info("邮件优先级：" + getPriority(msg));
			log.info("是否需要回执：" + isReplySign(msg));
			log.info("邮件大小：" + msg.getSize() * 1024 + "kb");
			StringBuffer content = new StringBuffer(30);
			getMailTextContent(msg, content);
			log.info("邮件正文：" + content);

			/*
			 * log.info(); boolean isContainerAttachment = isContainAttachment(msg);
			 * log.info("是否包含附件：" + isContainerAttachment); if (isContainerAttachment) {
			 * saveAttachment(msg, "D:\\data\\emailFile\\", msg.getFileName()); //保存附件 }
			 */
			log.info("------------------第" + msg.getMessageNumber() + "封邮件解析结束-------------------- ");
		}
	}

	/**
	 * 获得邮件发件人
	 * @param msg 邮件内容
	 * @return 姓名 <Email地址>
	 * @throws MessagingException
	 * @throws UnsupportedEncodingException
	 */
	public static String getFrom(MimeMessage msg) throws MessagingException, UnsupportedEncodingException {
		String from = "";
		Address[] froms = msg.getFrom();
		if (froms.length < 1)
			throw new MessagingException("没有发件人!");
		InternetAddress address = (InternetAddress) froms[0];
		String person = address.getPersonal();
		if (person != null) {
			person = MimeUtility.decodeText(person) + " ";
		}
		else {
			person = "";
		}
		from = person + "<" + address.getAddress() + ">";

		return from;
	}

	/**
	 * 获得邮件发送时间
	 * @param msg 邮件内容
	 * @return yyyy年mm月dd日 星期X HH:mm
	 * @throws MessagingException
	 */
	public static String getSentDate(MimeMessage msg, String pattern) throws MessagingException {
		Date receivedDate = msg.getSentDate();
		if (receivedDate == null)
			return "";

		if (pattern == null || "".equals(pattern))
			pattern = "yyyy年MM月dd日 E HH:mm ";

		return new SimpleDateFormat(pattern).format(receivedDate);
	}

	/**
	 * 判断邮件是否已读
	 * @param msg 邮件内容
	 * @return 如果邮件已读返回true,否则返回false
	 * @throws MessagingException
	 */
	public static boolean isSeen(MimeMessage msg) throws MessagingException {
		return msg.getFlags().contains(Flags.Flag.SEEN);
	}

	/**
	 * 判断邮件是否需要阅读回执
	 * @param msg 邮件内容
	 * @return 需要回执返回true,否则返回false
	 * @throws MessagingException
	 */
	public static boolean isReplySign(MimeMessage msg) throws MessagingException {
		boolean replySign = false;
		String[] headers = msg.getHeader("Disposition-Notification-To");
		if (headers != null)
			replySign = true;
		return replySign;
	}

	/**
	 * 获得邮件的优先级
	 * @param msg 邮件内容
	 * @return 1(High):紧急 3:普通(Normal) 5:低(Low)
	 * @throws MessagingException
	 */
	public static String getPriority(MimeMessage msg) throws MessagingException {
		String priority = "普通";
		String[] headers = msg.getHeader("X-Priority");
		if (headers != null) {
			String headerPriority = headers[0];
			if (headerPriority.contains("1") || headerPriority.contains("High"))
				priority = "紧急";
			else if (headerPriority.contains("5") || headerPriority.contains("Low"))
				priority = "低";
			else
				priority = "普通";
		}
		return priority;
	}

	/**
	 * 根据收件人类型，获取邮件收件人、抄送和密送地址。如果收件人类型为空，则获得所有的收件人
	 * <p>
	 * Message.RecipientType.TO 收件人
	 * </p>
	 * <p>
	 * Message.RecipientType.CC 抄送
	 * </p>
	 * <p>
	 * Message.RecipientType.BCC 密送
	 * </p>
	 * @param msg 邮件内容
	 * @param type 收件人类型
	 * @return 收件人1 <邮件地址1>, 收件人2 <邮件地址2>, ...
	 * @throws MessagingException
	 */
	public static String getReceiveAddress(MimeMessage msg, Message.RecipientType type) throws MessagingException {
		StringBuilder receiveAddress = new StringBuilder();
		Address[] addresss;
		if (type == null) {
			addresss = msg.getAllRecipients();
		}
		else {
			addresss = msg.getRecipients(type);
		}

		if (addresss == null || addresss.length < 1) {
			log.info("没有收件人");
			return "";
		}
		for (Address address : addresss) {
			InternetAddress internetAddress = (InternetAddress) address;
			receiveAddress.append(internetAddress.toUnicodeString()).append(",");
		}
		receiveAddress.deleteCharAt(receiveAddress.length() - 1); // 删除最后一个逗号
		return receiveAddress.toString();
	}

	/**
	 * 判断邮件中是否包含附件
	 * @return 存在附件返回true，不存在返回false
	 */
	public static boolean isContainAttachment(Part part) throws Exception {
		boolean flag = false;
		if (part.isMimeType("multipart/*")) {
			MimeMultipart multipart = (MimeMultipart) part.getContent();
			int partCount = multipart.getCount();
			for (int i = 0; i < partCount; i++) {
				BodyPart bodyPart = multipart.getBodyPart(i);
				String disp = bodyPart.getDisposition();
				if (disp != null && (disp.equalsIgnoreCase(Part.ATTACHMENT) || disp.equalsIgnoreCase(Part.INLINE))) {
					flag = true;
				}
				else if (bodyPart.isMimeType("multipart/*")) {
					flag = isContainAttachment(bodyPart);
				}
				else {
					String contentType = bodyPart.getContentType();
					if (contentType.contains("application")) {
						flag = true;
					}

					if (contentType.contains("name")) {
						flag = true;
					}
				}

				if (flag)
					break;
			}
		}
		else if (part.isMimeType("message/rfc822")) {
			flag = isContainAttachment((Part) part.getContent());
		}
		return flag;
	}

	/**
	 * 保存文件
	 * @param destDir 文件目录
	 * @param fileName 文件名
	 * @throws Exception 异常
	 */
	public static void saveAttachment(Part part, String destDir, String fileName) throws Exception {
		if (part.isMimeType("multipart/*")) {
			// 复杂体邮件
			Multipart multipart = (Multipart) part.getContent();
			// 复杂体邮件包含多个邮件体
			int partCount = multipart.getCount();
			for (int i = 0; i < partCount; i++) {
				// 获得复杂体邮件中其中一个邮件体
				BodyPart bodyPart = multipart.getBodyPart(i);
				// 迭代处理邮件体，直到附件为止
				String disp = bodyPart.getDisposition();
				String decodeName = decodeText(bodyPart.getFileName());
				decodeName = StringUtils.isEmpty(decodeName) ? fileName : decodeName;
				if (disp != null && (disp.equalsIgnoreCase(Part.ATTACHMENT) || disp.equalsIgnoreCase(Part.INLINE))) {
					saveFile(bodyPart.getInputStream(), destDir, decodeName);
				}
				else if (bodyPart.isMimeType("multipart/*")) {
					saveAttachment(bodyPart, destDir, fileName);
				}
				else {
					String contentType = bodyPart.getContentType();
					if (contentType.contains("name") || contentType.contains("application")) {
						saveFile(bodyPart.getInputStream(), destDir, decodeName);
					}
				}
			}
		}
		else if (part.isMimeType("message/rfc822")) {
			saveAttachment((Part) part.getContent(), destDir, fileName);
		}
	}

	/**
	 * 获得邮件文本内容
	 * @param part 邮件体
	 * @param content 存储邮件文本内容的字符串
	 * @throws MessagingException
	 * @throws IOException
	 */
	public static void getMailTextContent(Part part, StringBuffer content) throws MessagingException, IOException {
		// 如果是文本类型的附件，通过getContent方法可以取到文本内容，但这不是我们需要的结果，所以在这里要做判断
		boolean isContainTextAttach = part.getContentType().indexOf("name") > 0;
		if (part.isMimeType("text/*") && !isContainTextAttach) {
			content.append(part.getContent().toString());
		}
		else if (part.isMimeType("message/rfc822")) {
			getMailTextContent((Part) part.getContent(), content);
		}
		else if (part.isMimeType("multipart/*")) {
			Multipart multipart = (Multipart) part.getContent();
			int partCount = multipart.getCount();
			for (int i = 0; i < partCount; i++) {
				BodyPart bodyPart = multipart.getBodyPart(i);
				getMailTextContent(bodyPart, content);
			}
		}
	}

	/**
	 * 读取输入流中的数据保存至指定目录
	 * @param is 输入流
	 * @param fileName 文件名
	 * @param destDir 文件存储目录
	 */
	private static void saveFile(InputStream is, String destDir, String fileName) throws Exception {
		createEmptyDirectory(destDir);
		BufferedInputStream bis = new BufferedInputStream(is);
		BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(new File(destDir + fileName)));
		int len;
		while ((len = bis.read()) != -1) {
			bos.write(len);
			bos.flush();
		}
		bos.close();
		bis.close();
	}

	/**
	 * 创建一个空目录
	 */
	public static void createEmptyDirectory(String directoryPath) {
		File file = new File(directoryPath);
		if (!file.exists()) {
			file.mkdirs();
		}
	}

	/**
	 * 文本解码
	 */
	public static String decodeText(String encodeText) throws Exception {
		if (encodeText == null || "".equals(encodeText)) {
			return "";
		}
		else {
			return MimeUtility.decodeText(encodeText);
		}
	}

	/*
	 * public static void main(String[] args) {
	 *
	 * // 邮箱用户名和密码 final String username = "cheng_cd@murongtech.com"; final String
	 * password = "chengCd135";
	 *
	 * // 邮箱配置信息 Properties props = new Properties(); props.put("mail.smtp.host",
	 * "smtp.exmail.qq.com"); // 邮箱服务器地址 props.put("mail.smtp.port", "465"); // 邮箱服务器端口号
	 * props.put("mail.smtp.auth", "true"); // 启用身份验证 props.put("mail.smtp.ssl.enable",
	 * "true"); // 启用 SSL 加密
	 *
	 * // 创建 Session 对象，用于与邮箱服务器进行通信 Session session = Session.getInstance(props, new
	 * javax.mail.Authenticator() { protected PasswordAuthentication
	 * getPasswordAuthentication() { return new PasswordAuthentication(username,
	 * password); } });
	 *
	 * try { // 创建邮件对象 "cheng_cd@murongtech.com", "chengCd135"); Message message = new
	 * MimeMessage(session); message.setFrom(new
	 * InternetAddress("cheng_cd@murongtech.com")); // 设置发件人邮箱地址
	 * message.setRecipients(Message.RecipientType.TO,
	 * InternetAddress.parse("cheng_cd@murongtech.com")); // 设置收件人邮箱地址
	 * message.setSubject("Testing Subject"); // 设置邮件主题
	 * message.setText("Dear Mail Crawler," + "\n\n No spam to my email, please!"); //
	 * 设置邮件内容
	 *
	 * // 发送邮件 Transport.send(message);
	 *
	 * log.info("Done");
	 *
	 * } catch (MessagingException e) { throw new RuntimeException(e); } }
	 */

}